package day07_快速排序;

public class 快速排序3_version {

    public static void swap(int[] arr, int i, int j) {
        int tmp = arr[i];
        arr[i] = arr[j];
        arr[j] = tmp;
    }

    public static void quickSort3(int[] arr) {
        if (arr == null || arr.length < 2) {
            return;
        }
        process3(arr, 0, arr.length - 1);
    }

    public static void process3(int[] arr, int L, int R) {
        if (L >= R) {
            return;
        }
        swap(arr, L + (int) (Math.random() * (R - L + 1)), R);
        int[] equalArea = netherlandsFlag(arr, L, R);
        process3(arr, L, equalArea[0] - 1);
        process3(arr, equalArea[1] + 1, R);
    }

    // arr[L...R] 玩荷兰国旗问题的划分，以arr[R]做划分值
    // <arr[R]    ==arr[R]     > arr[R]
    private static int[] netherlandsFlag(int[] arr, int L, int R) {
        int less = L - 1; // < 区 右边界
        int more = R + 1; // > 区 左边界
        int cur = L;
        int num = arr[R];
        while (cur < more) { // 当前位置，不能和 >区的左边界撞上
            if (arr[cur] == num) {
                cur++;
            } else if (arr[cur] < num) {
                swap(arr, cur++, ++less);
            } else {
                swap(arr, cur, --more);
            }
        }
        return new int[]{less + 1, more - 1};
    }
}
